Python - function find list in list